from flask import abort
from flask import current_app, jsonify
from flask import g
from flask import render_template
from flask import request
from flask import session

from info import constants, db
from info.models import User, News, Comment
from info.utils.common import get_login_data
from info.utils.response_code import RET
from . import news_detail_blu


@news_detail_blu.route("/<int:news_id>")
@get_login_data
def news_detail(news_id):
    """

    详情页面和点击排行数据展示

    :param news_id:
    :return:
    """

    user = g.user
    # 点击排行数据展示
    try:
        news_click = News.query.order_by(News.clicks.desc()).limit(constants.CLICK_RANK_MAX_NEWS)
        # print(news_click.)
    except Exception as e:
        return jsonify(errno=RET.DATAERR, errmsg="数据库查询失败")

    # 查询详情页的详细新闻
    try:
        news = News.query.get(news_id)
    except Exception as e:
        current_app.logger.error(e)
        abort(404)

    if not news:
        # current_app.logger.error("这个新闻是none")
        abort(404)
    # 这个新闻的点击量加 一
    news.clicks += 1
    try:
        db.session.commit()
    except Exception as e:
        current_app.logger.error(e)
        db.session.rollback()

    # 展示收藏或者取消收藏  按钮的展示    考虑 用户登录了没有 登录了就展示收藏按钮或者已收藏按钮
    is_collected = False  # 定义变量没有收藏是false
    # 判断用户登录没有
    if user:
        # 判断用户有没有收藏这个新闻
        if news in user.collection_news:
            is_collected = True
    # 展示评论的数据
    try:
        comments = Comment.query.filter(Comment.news_id == news_id).order_by(Comment.create_time.desc()).all()
    except Exception as e:
        current_app.logger.error(e)

    comments_list = []
    for i in comments:
        comments_list.append(i.to_dict())

    # 获取作者对象
    author = news.user
    # 展示关注 或者取消关按钮
    is_followed = False  # 这个变量来标志用户是否关注作者
    # if 用户登录了：
    #     if 用户 关注了作者：
    #         告诉浏览器要显示 已关注 按钮

    # user.followed 获取用户关注了那些其他用户
    if user:
        if author in user.followed:
            is_followed = True

    user_like_comments = []
    for i in user.like_comments.all():
        user_like_comments.append(i.to_dict())

    data = {
        "user_info": user.to_dict() if user else None,  # user.to_dict() 可以通过它获取到用户的信息
        "news_click": news_click,
        "news": news.to_dict(),
        "is_collected": is_collected,
        "is_followed": is_followed,
        "comments": comments_list,
        "author": author.to_dict(),
        "user_like_comments": user_like_comments if user else []
    }
    return render_template("news/detail.html", data=data)


@news_detail_blu.route("/news_collect", methods=["POST"])
@get_login_data
def news_collect():
    """
    收藏和取消收藏功能函数
    需要的参数   news_id  新闻的id
    收藏   action   是collect就收藏   cancel_collect就取消收藏

    :return:
    """

    # 1获取登录信息判空
    user = g.user
    if not user:
        return jsonify(erron=RET.SERVERERR, errmasg="用户没有登录")
    # 2获取参数
    data_dict = request.json
    news_id = data_dict.get("news_id")
    action = data_dict.get("action")
    # 2.1判空
    if not all([news_id, action]):
        return jsonify(errno=RET.PARAMERR, errmsg="参数错误")
    if action not in ("collect", "cancel_collect"):
        return jsonify(errno=RET.PARAMERR, errmsg="参数错误")

    # 3 查询数据库 判断这个新闻是不是存在
    try:
        news = News.query.get(news_id)

    except Exception as e:
        current_app.logger.error(e)

    if not news:
        return jsonify(erron=RET.NODATA, errmsg="新闻不存在")

    if action == "collect":
        # 如果collect  就把这个新闻添加到用户的收藏
        user.collection_news.append(news)
    else:
        user.collection_news.remove(news)
    try:
        db.session.commit()
    except Exception as e:
        db.session.rollback()
        return jsonify(errno=RET.DBERR, errmsg="数据库操作失败")

    # 返回成功
    return jsonify(errno=RET.OK, errmsg="操作成功")


@news_detail_blu.route("/news_comment", methods=["POST"])
@get_login_data
def news_comment():
    """
    评论和回复评论功能
    需要的参数   news_id  新闻的id
    父评论的id
    评论的内容

    :return:
    """

    # 1获取登录信息判空
    user = g.user
    if not user:
        return jsonify(errno=RET.SERVERERR, errmasg="用户没有登录")
    # 2获取参数
    data_dict = request.json
    news_id = data_dict.get("news_id")
    comment_str = data_dict.get("comment")
    parent_id = data_dict.get("parent_id")

    # 3判空
    if not all([news_id, comment_str]):
        return jsonify(errno=RET.PARAMERR, errmsg="参数错误")

    # 3查询数据库 判断这个新闻是不是存在
    try:
        news = News.query.get(news_id)

    except Exception as e:
        current_app.logger.error(e)

    if not news:
        return jsonify(errno=RET.NODATA, errmsg="新闻不存在")

    # 4 把评论添加到数据库

    comment = Comment()
    comment.user_id = user.id
    comment.news_id = news_id
    comment.content = comment_str
    if parent_id:
        comment.parent_id = parent_id
    try:
        db.session.add(comment)
        db.session.commit()
    except Exception as e:
        current_app.logger.error(e)
        db.session.rollback()
        return jsonify(errno=RET.DBERR, errmsg="数据库操作失败")

    # 返回结果
    return jsonify(errno=RET.OK, errmsg="操作成功", data=comment.to_dict())


@news_detail_blu.route("/comment_like", methods=["POST"])
@get_login_data
def comment_like():
    """
       处理  点赞和取消点赞  的请求
       1、获取用户登录信息
       2、获取参数判空    评论id  action
       3、查询数据库，看评论是否存在
       4、添加点赞、 评论的like_count字段+1
       5、返回成功
       """

    # 1获取登录信息判空
    user = g.user
    if not user:
        return jsonify(erron=RET.SERVERERR, errmasg="用户没有登录")
    # 2获取参数
    data_dict = request.json
    comment_id = data_dict.get("comment_id")
    action = data_dict.get("action")

    # 2.1判空
    if not all([comment_id, action]):
        return jsonify(errno=RET.PARAMERR, errmsg="参数错误")
    if action not in ("add", "remove"):
        return jsonify(errno=RET.PARAMERR, errmsg="参数错误")

    # 3 查询数据库 判断这个评论是不是存在
    try:
        comment = Comment.query.get(comment_id)

    except Exception as e:
        current_app.logger.error(e)
        return jsonify(errno=RET.DBERR, errmsg="数据库查询错误")
    if not comment:
        return jsonify(erron=RET.NODATA, errmsg="评论不存在")

    if action == "add":
        # 如果add  就把这个评论添加到数据库
        user.like_comments.append(comment)
        comment.like_count += 1
    else:
        user.like_comments.remove(comment)
        comment.like_count -= 1
    try:
        db.session.commit()
    except Exception as e:
        db.session.rollback()
        return jsonify(errno=RET.DBERR, errmsg="数据库操作失败")

    # 返回成功
    return jsonify(errno=RET.OK, errmsg="操作成功")


@news_detail_blu.route("/followed_user", methods=["POST"])
@get_login_data
def followed_user():
    """
    处理 关注和 取消关注的请求 请求

    需要传的参数： author_id  action的值是follow就执行关注或者   unfollow 就取消关注的标志

    业务逻辑：
    1、获取登录信息
    2、获取参数
    3、判空
    4、查询数据库，判断作者是否存在
    5、判断action的值是  follow 还是  unfollow
    6、返回操作成功
    """
    # 1、获取登录信息
    user = g.user
    if not user:
        return jsonify(erron=RET.SERVERERR, errmasg="用户没有登录")
    # 2、获取参数
    data_dict = request.json
    action = data_dict.get("action")
    author_id = data_dict.get("user_id")

    if not all([author_id, action]):
        return jsonify(errno=RET.PARAMERR, errmsg="参数错误")
    if action not in ("follow", "unfollow"):
        return jsonify(errno=RET.PARAMERR, errmsg="参数错误")

    # 4、查询数据库，判断作者是否存在
    try:
        author = User.query.get(author_id)
    except Exception as e:
        current_app.logger.error(e)
        return jsonify(errno=RET.DBERR, errmsg="数据库查询错误")

    if action == "follow":
        # 执行关注
        user.followed.append(author)
    else:
        # 执行取消关注
        user.followed.remove(author)

    try:
        db.session.commit()
    except Exception as e:
        current_app.logger.error(e)
        db.session.rollback()
        return jsonify(errno=RET.DBERR, errmsg="数据库操作失败")

        # 6、返回操作成功
    return jsonify(errno=RET.OK, errmsg="操作成功")
